import torch

print("💻 PyTorch 是否安装成功：", torch.__version__)
print("📦 CUDA 是否可用：", torch.cuda.is_available())

if torch.cuda.is_available():
    print("🎉 CUDA 设备数：", torch.cuda.device_count())
    print("🚀 当前 CUDA 设备名称：", torch.cuda.get_device_name(torch.cuda.current_device()))
    print("🧠 当前设备 ID：", torch.cuda.current_device())
else:
    print("⚠️ 未检测到 CUDA 支持的 GPU，或 PyTorch 未安装 GPU 版本。")
